from 推荐定时脚本.RecService.rec.update_user import UpdateUserProfile
from 推荐定时脚本.RecService.rec.update_recall import updateRecall
from 推荐定时脚本.RecService.rec.update_video import updateVideoProfile


def update_user_profile():
    """
    更新用户画像
    :return:
    """
    print("===更新用户画像开始====")
    up = UpdateUserProfile()
    # 更新用户行为表
    up.update_user_action()
    # 更新用户画像
    up.update_user_info()
    print("=====更新用户画像结束===")
    return "ok"


def update_video_profile():
    """
    更新视频画像
    :return:
    """
    print("===更新视频画像开始====")
    vp = updateVideoProfile()
    vp.update_video_profile()
    vp.update_inverted_table()
    print("=====更新视频画像结束===")


def update_video_similar():
    print("===更新视频相似度开始====")
    vp = updateVideoProfile()
    vp.update_video_similar()
    print("===更新视频相似度开始====")


def update_recall():
    """
    更新用户的召回集
    :return:
    """
    print("更新召回开始")
    ur = updateRecall()
    # 更新用户历史召回推荐结果
    ur.update_user_history_recall()
    # 更新用户画像的召回结果
    ur.update_profile_reacall()
    print("====更新召回结束=======")
    return "ok"
